package com.llbnk.srb.core.controller.admin;

import com.alibaba.excel.EasyExcel;
import com.llbnk.common.exception.BusinessException;
import com.llbnk.common.result.R;
import com.llbnk.common.result.ResponseEnum;
import com.llbnk.srb.core.pojo.dto.ExcelDictDTO;
import com.llbnk.srb.core.pojo.entity.Dict;
import com.llbnk.srb.core.service.DictService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.util.List;

/**
 * @author shkstart
 * @create 2022-04-10 9:57
 */

@Api(tags = "数据字典管理")
@RestController
@RequestMapping("/admin/core/dict")
@Slf4j
//@CrossOrigin
public class AdminDictController {
    /**
     * excel导入后端流程：
     * 前端调用后端import接口，在batchImport实现，在方法中拿到file文件拿到inputStream
     * 然后传递给Service层做业务（调用的还是DictServiceImpl，DictService的实现类）
     * 在service（DictServiceImpl）中创建了一个对象监听器ExcelDictDTOListener()
     * 通过监听器创建我们进行了Excel的读，通过doRead()去执行读
     * 并希望自动的将读出来的数据封装到ExcelDictDTO里面
     * 具体的封装过程是在ExcelDictDTOListener监听器中实现的，重写方法调用mapper层完成数据的存储
     */
    @Resource
    private DictService dictService;


    @ApiOperation("Excel数据的批量导入")
    @PostMapping("/import")
    public R batchImport(
            @ApiParam(value = "Excel数据字典文件",required = true)
            @RequestParam("file")MultipartFile file
            ){
        try {
            InputStream inputStream = file.getInputStream();
            dictService.importData(inputStream);
            return R.ok().message("数据字典数据批量导入成功");
        } catch (Exception e) {
            //BusinessException(ResponseEnum resultCodeEnum, Throwable cause)
            e.printStackTrace();
            throw new BusinessException(ResponseEnum.UPLOAD_ERROR,e);
        }
    }

    /**
     * 文件下载（失败了会返回一个有部分数据的Excel）
     * <p>
     * 1. 创建excel对应的实体对象
     * <p>
     * 2. 设置返回的 参数
     * <p>
     * 3. 直接写，这里注意，finish的时候会自动关闭OutputStream,当然你外面再关闭流问题不大
     */
    @ApiOperation("Excel数据的导出")
    @GetMapping("/export")
    public void export(HttpServletResponse response) throws IOException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("mydict", "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), ExcelDictDTO.class)
                .sheet("数据字典")
                .doWrite(dictService.listDictData());//写的（从后端数据库中下载数据到浏览器中）业务交给Service层来做
        //doWrite写的参数并不是对应数据库表中的实体类（也就是entity中的Dict实体类）
        //而是作为easyExcel这个封装的实体（dto中的ExcelDictDTO）
    }

    /**
     * 树形数据的加载方案：
     * 延迟加载
     * 先加载一级大的数据，当点开这个列表的时候，才回去加载里面小的列表
     */
    @ApiOperation("根据上级id获取子节点数据列表")
    @GetMapping("/listByParentId/{parentId}")
    public R listByParentId(
            @ApiParam(value = "上级节点id", required = true)
            @PathVariable Long parentId){
        List<Dict> dictList = dictService.listByParentId(parentId);
        return R.ok().data("list",dictList);
    }

}
